from kyb_caps import driver

driver.get_screenshot_as_file(r'login1.png')
driver.save_screenshot(r'login2.png')

driver.quit()
